package List;

/**
 * 203.移除链表元素
 * 给你一个链表的头节点 head 和一个整数 val ，请你删除链表中所有满足 Node.val == val 的节点，并返回 新的头节点 。
 */
public class removeElements {

    public class ListNode {
        int val;
        ListNode next;

        ListNode() {
        }

        ListNode(int val) {
            this.val = val;
        }

        ListNode(int val, ListNode next) {
            this.val = val;
            this.next = next;
        }
    }

    /**
     * 直接在list上进行操作
     * @param head
     * @param val
     * @return
     */
    public ListNode removeElements(ListNode head, int val) {
        if (head == null) return null;

        ListNode cru = head;
        while (cru != null && cru.next != null) {
            if (cru.next.val == val) {
                cru.next = cru.next.next;
            } else {
                cru = cru.next;
            }
        }
        return head;
    }

    /**
     * 设置虚拟头节点
     * @param head
     * @param val
     * @return
     */
    public ListNode removeElements2(ListNode head, int val) {
        ListNode dummy  = new ListNode();
        dummy.next = head;

        ListNode cur = dummy ;
        while (cur.next != null) {
            if (cur.next.val == val) {
                cur.next = cur.next.next;
            } else {
                cur = cur.next;
            }
        }
        return dummy.next;
    }


}
